Replace while loop with lodash